/*
Copyright (C) 1996-1997 Id Software, Inc.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 3
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "quakedef.h"
#include "menu.shared.h"


void Key_SetDest (keydest_t dest)
{
	if (dest != key_dest)
	{
		Host_RearmTimers ();
		key_dest = dest;
	}
}


/*

key up events are sent even if in console mode

*/


#define      HISTORY_FILE_NAME   "id1/quake_history.txt"

#define      MAXCMDLINE   256
#define      CMDLINES   32

char   key_lines[CMDLINES][MAXCMDLINE];

int		key_linepos;
int		shift_down = FALSE;
int		key_lastpress;

int		edit_line = 0;
int		history_line = 0;

keydest_t	key_dest;

int		key_count;			// incremented every key event

char	*keybindings[256];
BOOL	consolekeys[256];	// if TRUE, can't be rebound while in console
BOOL	menubound[256];	// if TRUE, can't be rebound while in menu
int		keyshift[256];		// key to map to if shift held down in console
int		key_repeats[256];	// if > 1, it is autorepeating
BOOL	keydown[256];

typedef struct
{
	const char	*name;
	int		keynum;
} keyname_t;

#define DEFINE_KEY_NAME(kn) {&(#kn)[2], kn}

keyname_t keynames[] =
{
	DEFINE_KEY_NAME (K_TAB),
	DEFINE_KEY_NAME (K_ENTER),
	DEFINE_KEY_NAME (K_ESCAPE),
	DEFINE_KEY_NAME (K_SPACE),
	DEFINE_KEY_NAME (K_BACKSPACE),
	DEFINE_KEY_NAME (K_UPARROW),
	DEFINE_KEY_NAME (K_DOWNARROW),
	DEFINE_KEY_NAME (K_LEFTARROW),
	DEFINE_KEY_NAME (K_RIGHTARROW),

	DEFINE_KEY_NAME (K_ALT),
	DEFINE_KEY_NAME (K_CTRL),
	DEFINE_KEY_NAME (K_SHIFT),

	DEFINE_KEY_NAME (K_F1),
	DEFINE_KEY_NAME (K_F2),
	DEFINE_KEY_NAME (K_F3),
	DEFINE_KEY_NAME (K_F4),
	DEFINE_KEY_NAME (K_F5),
	DEFINE_KEY_NAME (K_F6),
	DEFINE_KEY_NAME (K_F7),
	DEFINE_KEY_NAME (K_F8),
	DEFINE_KEY_NAME (K_F9),
	DEFINE_KEY_NAME (K_F10),
	DEFINE_KEY_NAME (K_F11),
	DEFINE_KEY_NAME (K_F12),

	DEFINE_KEY_NAME (K_INS),
	DEFINE_KEY_NAME (K_DEL),
	DEFINE_KEY_NAME (K_PGDN),
	DEFINE_KEY_NAME (K_PGUP),
	DEFINE_KEY_NAME (K_HOME),
	DEFINE_KEY_NAME (K_END),

	DEFINE_KEY_NAME (K_MOUSE1),
	DEFINE_KEY_NAME (K_MOUSE2),
	DEFINE_KEY_NAME (K_MOUSE3),

	DEFINE_KEY_NAME (K_JOY1),
	DEFINE_KEY_NAME (K_JOY2),
	DEFINE_KEY_NAME (K_JOY3),
	DEFINE_KEY_NAME (K_JOY4),

	DEFINE_KEY_NAME (K_AUX1),
	DEFINE_KEY_NAME (K_AUX2),
	DEFINE_KEY_NAME (K_AUX3),
	DEFINE_KEY_NAME (K_AUX4),
	DEFINE_KEY_NAME (K_AUX5),
	DEFINE_KEY_NAME (K_AUX6),
	DEFINE_KEY_NAME (K_AUX7),
	DEFINE_KEY_NAME (K_AUX8),
	DEFINE_KEY_NAME (K_AUX9),
	DEFINE_KEY_NAME (K_AUX10),
	DEFINE_KEY_NAME (K_AUX11),
	DEFINE_KEY_NAME (K_AUX12),
	DEFINE_KEY_NAME (K_AUX13),
	DEFINE_KEY_NAME (K_AUX14),
	DEFINE_KEY_NAME (K_AUX15),
	DEFINE_KEY_NAME (K_AUX16),
	DEFINE_KEY_NAME (K_AUX17),
	DEFINE_KEY_NAME (K_AUX18),
	DEFINE_KEY_NAME (K_AUX19),
	DEFINE_KEY_NAME (K_AUX20),
	DEFINE_KEY_NAME (K_AUX21),
	DEFINE_KEY_NAME (K_AUX22),
	DEFINE_KEY_NAME (K_AUX23),
	DEFINE_KEY_NAME (K_AUX24),
	DEFINE_KEY_NAME (K_AUX25),
	DEFINE_KEY_NAME (K_AUX26),
	DEFINE_KEY_NAME (K_AUX27),
	DEFINE_KEY_NAME (K_AUX28),
	DEFINE_KEY_NAME (K_AUX29),
	DEFINE_KEY_NAME (K_AUX30),
	DEFINE_KEY_NAME (K_AUX31),
	DEFINE_KEY_NAME (K_AUX32),

	DEFINE_KEY_NAME (K_PAUSE),

	DEFINE_KEY_NAME (K_MWHEELUP),
	DEFINE_KEY_NAME (K_MWHEELDOWN),

	{"SEMICOLON", ';'},	// because a raw semicolon seperates commands

	{NULL, 0}
};

/*
==============================================================================

			LINE TYPING INTO THE CONSOLE

==============================================================================
*/

key_autotab_t key_autotab_cvar = {0, 0, 0, TRUE};
key_autotab_t key_autotab_fs = {0, 0, 0, TRUE};


void Key_ResetAutoTab (key_autotab_t *at, BOOL begin)
{
	// reset for a new match cycle
	at->FirstMatch = 0;
	at->NumMatches = 0;
	at->CurrentItem = 0;
	at->StartMatch = begin;
}


void Key_AddToAutoTab (key_autotab_t *at, int item)
{
	// store the index of the first item
	if (!at->NumMatches)
		at->FirstMatch = item;

	// count the matched items
	at->NumMatches++;
}


int Key_RunAutoTab (key_autotab_t *at)
{
	// get the current item
	int ThisItem = at->FirstMatch + at->CurrentItem;

	// go to the next match
	at->CurrentItem = (at->CurrentItem + 1) % at->NumMatches;

	// and return the one we got
	return ThisItem;
}


void Key_UpdateCompleteLine (char *cmd, BOOL addspace)
{
	Q_strcpy (key_lines[edit_line] + 1, cmd);
	key_linepos = Q_strlen (cmd) + 1;

	if (addspace)
	{
		key_lines[edit_line][key_linepos] = ' ';
		key_linepos++;
	}

	key_lines[edit_line][key_linepos] = 0;
}


/*
====================
Key_Console

Interactive line editing and console scrollback
====================
*/
void Key_Console (int key)
{
	// check for tab first and separately so that we can handle autotab setting and resetting properly
	if (key == K_TAB)
	{
		char *cmd = NULL;

		if ((cmd = FS_AutoCompleteList (&key_autotab_fs, key_lines[edit_line] + 1)) != NULL)
		{
			Key_UpdateCompleteLine (cmd, FALSE);
			return;
		}

		// command completion
		if ((cmd = Cvar_CompleteVariable (&key_autotab_cvar, key_lines[edit_line] + 1)) != NULL)
		{
			Key_UpdateCompleteLine (cmd, TRUE);
			return;
		}
	}

	// reset the autotab lists
	Key_ResetAutoTab (&key_autotab_fs, TRUE);
	Key_ResetAutoTab (&key_autotab_cvar, TRUE);

	// now check other keys
	switch (key)
	{
	case K_ENTER:
		Cbuf_AddText (key_lines[edit_line] + 1);	// skip the >
		Cbuf_AddText ("\n");
		Con_Printf (PRINT_DEFAULT, "%s\n", key_lines[edit_line]);
		edit_line = (edit_line + 1) & (CMDLINES - 1);
		history_line = edit_line;
		key_lines[edit_line][0] = ']';
		key_linepos = 1;

		// force an update, because the command may take some time
		if (cls.state == ca_disconnected)
			SCR_UpdateScreen (0);

		return;

	case K_BACKSPACE:
	case K_LEFTARROW:
		if (key_linepos > 1)
			key_linepos--;

		return;

	case K_UPARROW:
		do
		{
			history_line = (history_line - 1) & (CMDLINES - 1);
		} while (history_line != edit_line && !key_lines[history_line][1]);

		if (history_line == edit_line)
			history_line = (edit_line + 1) & (CMDLINES - 1);

		Q_strcpy (key_lines[edit_line], key_lines[history_line]);
		key_linepos = Q_strlen (key_lines[edit_line]);

		return;

	case K_DOWNARROW:
		if (history_line == edit_line) return;

		do
		{
			history_line = (history_line + 1) & (CMDLINES - 1);
		} while (history_line != edit_line && !key_lines[history_line][1]);

		if (history_line == edit_line)
		{
			key_lines[edit_line][0] = ']';
			key_linepos = 1;
		}
		else
		{
			Q_strcpy (key_lines[edit_line], key_lines[history_line]);
			key_linepos = Q_strlen (key_lines[edit_line]);
		}

		return;

	case K_PGUP:
	case K_MWHEELUP:
		con_backscroll += 2;

		if (con_backscroll > con_totallines - (vid.Height >> 3) - 1)
			con_backscroll = con_totallines - (vid.Height >> 3) - 1;

		return;

	case K_PGDN:
	case K_MWHEELDOWN:
		con_backscroll -= 2;

		if (con_backscroll < 0)
			con_backscroll = 0;

		return;

	case K_HOME:
		con_backscroll = con_totallines - (vid.Height >> 3) - 1;
		return;

	case K_END:
		con_backscroll = 0;
		return;

	default:
		// check for non printable
		if (key < 32 || key > 127)
			return;
		else break;
	}

	if (key_linepos < MAXCMDLINE - 1)
	{
		key_lines[edit_line][key_linepos] = key;
		key_linepos++;
		key_lines[edit_line][key_linepos] = 0;
	}
}

//============================================================================

#define MAX_CHAT_BUFFER	32

char chat_buffer[MAX_CHAT_BUFFER];
BOOL team_message = FALSE;

void Key_Message (int key)
{
	static int chat_bufferlen = 0;

	if (key == K_ENTER)
	{
		if (team_message)
			Cbuf_AddText ("say_team \"");
		else
			Cbuf_AddText ("say \"");

		Cbuf_AddText (chat_buffer);
		Cbuf_AddText ("\"\n");

		Key_SetDest (key_game);
		chat_bufferlen = 0;
		chat_buffer[0] = 0;
		return;
	}

	if (key == K_ESCAPE)
	{
		Key_SetDest (key_game);
		chat_bufferlen = 0;
		chat_buffer[0] = 0;
		return;
	}

	if (key < 32 || key > 127)
		return;	// non printable

	if (key == K_BACKSPACE)
	{
		if (chat_bufferlen)
		{
			chat_bufferlen--;
			chat_buffer[chat_bufferlen] = 0;
		}

		return;
	}

	// reserving an extra space for the null-termination
	if (chat_bufferlen < (MAX_CHAT_BUFFER - 1))
	{
		chat_buffer[chat_bufferlen++] = key;
		chat_buffer[chat_bufferlen] = 0;
	}
}

//============================================================================


/*
===================
Key_StringToKeynum

Returns a key number to be used to index keybindings[] by looking at
the given string.  Single ascii characters return themselves, while
the K_* names are matched up.
===================
*/
int Key_StringToKeynum (char *str)
{
	if (!str || !str[0])
		return -1;

	if (!str[1])
		return str[0];

	for (keyname_t *kn = keynames; kn->name; kn++)
	{
		if (!stricmp (str, kn->name))
			return kn->keynum;
	}

	return -1;
}

/*
===================
Key_KeynumToString

Returns a string (either a single ascii char, or a K_* name) for the
given keynum.
FIXME: handle quote special (general escape sequence?)
===================
*/
const char *Key_KeynumToString (int keynum)
{
	static	char	tinystr[2];

	if (keynum == -1)
		return "<KEY NOT FOUND>";

	if (keynum > 32 && keynum < 127)
	{
		// printable ascii
		tinystr[0] = keynum;
		tinystr[1] = 0;
		return tinystr;
	}

	for (keyname_t *kn = keynames; kn->name; kn++)
		if (keynum == kn->keynum)
			return kn->name;

	return "<UNKNOWN KEYNUM>";
}


/*
===================
Key_SetBinding
===================
*/
void Key_SetBinding (int keynum, char *binding)
{
	if (keybindings[keynum & 255])
	{
		zoneHeap->Free (keybindings[keynum & 255]);
		keybindings[keynum & 255] = NULL;
	}

	keybindings[keynum & 255] = CopyString (binding);
}


/*
===================
Key_Unbind_f
===================
*/
void Key_Unbind_f (void)
{
	int		b;

	if (Cmd_Argc () != 2)
	{
		Con_Printf (PRINT_DEFAULT, "unbind <key> : remove commands from a key\n");
		return;
	}

	b = Key_StringToKeynum (Cmd_Argv (1));

	if (b == -1)
	{
		Con_Printf (PRINT_DEFAULT, "\"%s\" isn't a valid key\n", Cmd_Argv (1));
		return;
	}

	Key_SetBinding (b, "");
}

void Key_Unbindall_f (void)
{
	int		i;

	for (i = 0; i < 256; i++)
		if (keybindings[i])
			Key_SetBinding (i, "");
}


/*
===================
Key_Bind_f
===================
*/
void Key_Bind_f (void)
{
	int			i, c, b;
	char		cmd[1024];

	c = Cmd_Argc ();

	if (c != 2 && c != 3)
	{
		Con_Printf (PRINT_DEFAULT, "bind <key> [command] : attach a command to a key\n");
		return;
	}

	b = Key_StringToKeynum (Cmd_Argv (1));

	if (b == -1)
	{
		Con_Printf (PRINT_DEFAULT, "\"%s\" isn't a valid key\n", Cmd_Argv (1));
		return;
	}

	if (c == 2)
	{
		if (keybindings[b])
			Con_Printf (PRINT_DEFAULT, "\"%s\" = \"%s\"\n", Cmd_Argv (1), keybindings[b]);
		else Con_Printf (PRINT_DEFAULT, "\"%s\" is not bound\n", Cmd_Argv (1));

		return;
	}

	// copy the rest of the command line
	cmd[0] = 0;		// start out with a null string

	for (i = 2; i < c; i++)
	{
		if (i > 2)
			strcat (cmd, " ");

		strcat (cmd, Cmd_Argv (i));
	}

	Key_SetBinding (b, cmd);
}

/*
============
Key_WriteBindings

Writes lines containing "bind key value"
============
*/
void Key_WriteBindings (FILE *f)
{
	int		i;

	// begin with a fresh set
	fprintf (f, "unbindall\n");

	for (i = 0; i < 256; i++)
	{
		if (!keybindings[i]) continue;
		if (!keybindings[i][0]) continue;

		fprintf (f, "bind \"%s\" \"%s\"\n", Key_KeynumToString (i), keybindings[i]);
	}
}


void History_Init (void)
{
	int i, c;
	FILE *hf;

	for (i = 0; i < CMDLINES; i++)
	{
		key_lines[i][0] = ']';
		key_lines[i][1] = 0;
	}

	key_linepos = 1;

	if ((hf = fopen (HISTORY_FILE_NAME, "rt")))
	{
		do
		{
			i = 1;

			do
			{
				c = fgetc (hf);
				key_lines[edit_line][i++] = c;
			} while (c != '\n' && c != EOF && i < MAXCMDLINE);

			key_lines[edit_line][i - 1] = 0;
			edit_line = (edit_line + 1) & (CMDLINES - 1);
		} while (c != EOF && edit_line < CMDLINES);

		fclose (hf);

		history_line = edit_line = (edit_line - 1) & (CMDLINES - 1);
		key_lines[edit_line][0] = ']';
		key_lines[edit_line][1] = 0;
	}
}


void History_Shutdown (void)
{
	int i;
	FILE *hf;

	if ((hf = fopen (HISTORY_FILE_NAME, "wt")))
	{
		i = edit_line;

		do
		{
			i = (i + 1) & (CMDLINES - 1);
		} while (i != edit_line && !key_lines[i][1]);

		do
		{
			// fprintf(hf, "%s\n", wcs2str(key_lines[i] + 1)); // Baker: I commented this line out because byte colored text isn't a feature in most ordinary engines
			fprintf (hf, "%s\n", key_lines[i] + 1);
			i = (i + 1) & (CMDLINES - 1);
		} while (i != edit_line && key_lines[i][1]);

		fclose (hf);
	}
}


/*
===================
Key_Init
===================
*/
void Key_Init (void)
{
	int		i;

	History_Init ();

	key_linepos = 1;

	// init ascii characters in console mode
	for (i = 32; i < 128; i++)
		consolekeys[i] = TRUE;

	consolekeys[K_ENTER] = TRUE;
	consolekeys[K_TAB] = TRUE;
	consolekeys[K_LEFTARROW] = TRUE;
	consolekeys[K_RIGHTARROW] = TRUE;
	consolekeys[K_UPARROW] = TRUE;
	consolekeys[K_DOWNARROW] = TRUE;
	consolekeys[K_BACKSPACE] = TRUE;
	consolekeys[K_PGUP] = TRUE;
	consolekeys[K_PGDN] = TRUE;
	consolekeys[K_SHIFT] = TRUE;
	consolekeys[K_MWHEELUP] = TRUE;
	consolekeys[K_MWHEELDOWN] = TRUE;
	consolekeys['`'] = FALSE;
	consolekeys['~'] = FALSE;

	for (i = 0; i < 256; i++)
		keyshift[i] = i;

	for (i = 'a'; i <= 'z'; i++)
		keyshift[i] = i - 'a' + 'A';

	keyshift['1'] = '!';
	keyshift['2'] = '@';
	keyshift['3'] = '#';
	keyshift['4'] = '$';
	keyshift['5'] = '%';
	keyshift['6'] = '^';
	keyshift['7'] = '&';
	keyshift['8'] = '*';
	keyshift['9'] = '(';
	keyshift['0'] = ')';
	keyshift['-'] = '_';
	keyshift['='] = '+';
	keyshift[','] = '<';
	keyshift['.'] = '>';
	keyshift['/'] = '?';
	keyshift[';'] = ':';
	keyshift['\''] = '"';
	keyshift['['] = '{';
	keyshift[']'] = '}';
	keyshift['`'] = '~';
	keyshift['\\'] = '|';

	menubound[K_ESCAPE] = TRUE;

	for (i = 0; i < 12; i++)
		menubound[K_F1 + i] = TRUE;

	// register our functions
	Cmd_AddCommand ("bind", Key_Bind_f);
	Cmd_AddCommand ("unbind", Key_Unbind_f);
	Cmd_AddCommand ("unbindall", Key_Unbindall_f);
}

/*
===================
Key_Event

Called by the system between frames for both key up and key down events
Should NOT be called during an interrupt!
===================
*/
void Key_Event (int key, BOOL down)
{
	char	*kb;
	char	cmd[1024];

	keydown[key] = down;

	if (!down)
		key_repeats[key] = 0;

	key_lastpress = key;
	key_count++;

	if (key_count <= 0)
	{
		// just catching keys for Con_NotifyBox
		return;
	}

	// update auto-repeat status
	if (down)
	{
		key_repeats[key]++;

		if (key != K_BACKSPACE && key != K_PAUSE && key_repeats[key] > 1 && key_dest == key_game)
		{
			// ignore most autorepeats
			return;
		}

		if (key >= 200 && !keybindings[key])
			Con_Printf (PRINT_DEVELOPER, "%s is unbound, hit F4 to set.\n", Key_KeynumToString (key));
	}

	if (key == K_SHIFT)
		shift_down = down;

	// handle escape specialy, so the user can never unbind it
	if (key == K_ESCAPE)
	{
		if (!down)
			return;

		switch (key_dest)
		{
		case key_message:
			Key_Message (key);
			break;

		case key_menu:
			menus[m_state]->Key (key);
			break;

		case key_game:
		case key_console:
			M_ToggleMenu_f ();
			break;

		default:
			Sys_Error ("Bad key_dest");
		}

		return;
	}

	// key up events only generate commands if the game key binding is
	// a button command (leading + sign).  These will occur even in console mode,
	// to keep the character from continuing an action started before a console
	// switch.  Button commands include the kenum as a parameter, so multiple
	// downs can be matched with ups
	if (!down)
	{
		kb = keybindings[key];

		if (kb && kb[0] == '+')
		{
			sprintf (cmd, "-%s %i\n", kb + 1, key);
			Cbuf_AddText (cmd);
		}

		if (keyshift[key] != key)
		{
			kb = keybindings[keyshift[key]];

			if (kb && kb[0] == '+')
			{
				sprintf (cmd, "-%s %i\n", kb + 1, key);
				Cbuf_AddText (cmd);
			}
		}

		return;
	}

	// during demo playback, most keys bring up the main menu
	if (cls.demoplayback && down && consolekeys[key] && key_dest == key_game)
	{
		M_ToggleMenu_f ();
		return;
	}

	// if not a consolekey, send to the interpreter no matter what mode is
	if ((key_dest == key_menu && menubound[key]) ||
		(key_dest == key_console && !consolekeys[key]) ||
		(key_dest == key_game && (!con_forcedup || !consolekeys[key])))
	{
		kb = keybindings[key];

		if (kb)
		{
			if (kb[0] == '+')
			{
				// button commands add keynum as a parm
				sprintf (cmd, "%s %i\n", kb, key);
				Cbuf_AddText (cmd);
			}
			else
			{
				Cbuf_AddText (kb);
				Cbuf_AddText ("\n");
			}
		}

		return;
	}

	if (!down)
		return;		// other systems only care about key down events

	if (shift_down)
	{
		key = keyshift[key];
	}

	switch (key_dest)
	{
	case key_message:
		Key_Message (key);
		break;

	case key_menu:
		menus[m_state]->Key (key);
		break;

	case key_game:
	case key_console:
		Key_Console (key);
		break;

	default:
		Sys_Error ("Bad key_dest");
	}
}


void Key_ClearButton (int key)
{
	if (keydown[key])
	{
		Key_Event (key, FALSE);
		keydown[key] = FALSE;
	}

	// clear the repeat state
	key_repeats[key] = 0;
}


/*
===================
Key_ClearStates
===================
*/
void Key_ClearStates (void)
{
	for (int i = 0; i < 256; i++)
	{
		// send an up event for each key that's currently down, to make sure the server clears them all
		Key_ClearButton (i);
	}
}

